import numpy as np

# 创建测试数据（包含异常值以便演示）
temps = np.array([
    [24.1, 18.4, 29.8, 14.3, 22.7, 31.4, 16.7],
    [19.8, 27.6, 33.1, 20.4, 17.8, 25.3, 28.7],
    [22.3, 16.7, np.nan, 30.1, 24.5, 19.8, 21.3],
    [31.2, 26.7, 18.4, 22.3, 29.8, 17.6, 25.1],
    [20.4, 23.7, 19.8, 30.1, np.inf, 22.3, 28.7],
    [27.6, np.inf, 24.5, 19.8, 23.4, 31.2, 26.7],
    [-5.0, 22.3, 30.1, 25.6, 19.8, 24.5, 36.0]  # 包含明显异常值
])

# (1) 提取城市3数据
city3 = temps[2, :]
print("城市3气温：", city3)

# (2) 提取前3天数据
first3_days = temps[:, 0:3]
print("前三天温度：\n", first3_days)

# (3) 筛选异常温度
abnormal = temps[(temps < 0) | (temps > 35)]
print("异常温度：", abnormal)

# (4) 沿城市维度排序（每天的温度）
sorted_temps = np.sort(temps, axis=1)
print("排序后气温：\n", sorted_temps)